# 使用python openpyxl 库读取指定文件夹下的(C:\Users\dwdar\Desktop\test1\data) 所有文件,读取所有工作簿的Sheet1工作表中的数据,读取第14行至第33行,第1列至33行数据,
# 先将数据打印到屏幕

import os
from openpyxl import load_workbook
import numpy as np

# 指定文件夹路径
folder_path = 'data'

# 遍历文件夹中的所有文件
year_0 = []
year_1 = []
year_2 = []
year_3 = []
year_4 = []
year_5 = []
year_6 = []
year_7 = []
year_8 = []
year_9 = []
year_10 = []
year_11 = []
year_12 = []
year_13 = []
year_14 = []
year_15 = []
year_16 = []
year_17 = []
year_18 = []
year_19 = []



for filename in os.listdir(folder_path):
    if filename.endswith('.xlsx') or filename.endswith('.xls'):  # 只处理 Excel 文件
        file_path = os.path.join(folder_path, filename)
        
        # 加载工作簿
        wb = load_workbook(file_path)
        
        # 选择 Sheet1
        sheet = wb['Sheet1']
        
        # 读取第8行至第33行，第1列至第36列的数据
        data = []
        for row in sheet.iter_rows(min_row=14, max_row=33, min_col=1, max_col=33):
            row_data = list()
            for cell in row:
                if isinstance(cell.value,int):
                    row_data.append(cell.value) 
            data.append(row_data)

        # 添加并计算
        year_0.append(data[0])
        year_1.append(data[1])
        year_2.append(data[2])
        year_3.append(data[3])
        year_4.append(data[4])
        year_5.append(data[5])
        year_6.append(data[6])
        year_7.append(data[7])
        year_8.append(data[8])
        year_9.append(data[9])
        year_10.append(data[10])
        year_11.append(data[11])
        year_12.append(data[12])
        year_13.append(data[13])
        year_14.append(data[14])
        year_15.append(data[15])
        year_16.append(data[16])
        year_17.append(data[17])
        year_18.append(data[18])
        year_19.append(data[19])


        # 打印数据到屏幕
        print(f"文件: {filename}")
        print(np.sum(year_19, axis = 0))
        # for item in data:
        #     print(item)
        print('\n')


